#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <unistd.h>
#include <ctype.h>
#include <errno.h>
#include <sys/socket.h>
#include <arpa/inet.h>
#include <pthread.h>

void sys_err(const char *str){
	perror(str);
	exit(1);
}

int main(int argc, char *argv[]){
	int fd,cfd;
	struct sockaddr_in serv_addr, clit_addr;
	int ret,i,clit_addr_len;
	char buf[BUFSIZ],client_IP[1024];
	pid_t pid;	// 定义管道描述符

	// 1.创建socket套接字，AF_INET表示IPv4，
	fd = socket(AF_INET,SOCK_STREAM,0);
	if(fd == -1){
		printf("套接字创建失败\n");
		return -1;
	}else{
		printf("套接字创建成功\n");
	}
	
	// 2.使用bind函数进行绑定
	serv_addr.sin_family = AF_INET;
	serv_addr.sin_port = htons(8888);
	serv_addr.sin_addr.s_addr = htonl(INADDR_ANY);

	ret = bind(fd,(struct sockaddr *)&serv_addr,sizeof(serv_addr));

	if(ret  == 0){
		printf("IP和端口绑定成功\n");
	}else{
		printf("bind函数绑定失败\n");
			exit(-1);
	}

	// 3.使用listen()函数设置监听上限
	if( listen(fd,12) == -1){
		printf("listen函数调用失败\n");
			exit(-1);
	}
	
	// 4.使用accept()函数阻塞，并监听客户端的链接
	while(1){
		clit_addr_len = sizeof(clit_addr);
		cfd = accept(fd, (struct sockaddr *)&clit_addr, &clit_addr_len);
		if(cfd == -1){
			printf("accept函数调用失败\n");	
				exit(-1);
		}else{
			printf("accept函数调用成功\n");
		}
		
		// 打印客户端的IP地址和端口号
		printf("IP:%s, Port:%d\n", inet_ntop(AF_INET, &clit_addr.sin_addr.s_addr, client_IP, sizeof(client_IP)), ntohs(clit_addr.sin_port));

		// 创建子进程，在子进程中处理
		pid = fork();
		if(pid == 0){
			printf("进入子进程!\n");
			close(fd);
			// 5.使用read()和write()函数进行读写
			while(1){
				//printf("显示数据： ");
				bzero(buf,sizeof(buf));	// 先清空缓冲区
				ret = read(cfd, buf, sizeof(buf));
				if(ret == -1){
					printf("读取失败!\n");
					exit(-1);
				}
				if(ret != 0) printf("%s\n",buf);
				// write(STDOUT_FILENO,buf,ret); 
				for(i = 0; i < ret; ++i){
					buf[i] = toupper(buf[i]);
				}
		
				if(write(cfd, buf, ret) == -1){
					printf("写入失败!\n");
					exit(-1);
				}
			}
			close(cfd);
		}
		else if(pid > 0){
			// 在父进程中关闭套接字，准备下一个客户端的连接
			printf("子进程退出!\n");
			close(cfd);
		}
	}
	
	// 6.退出连接
	close(fd);
	//close(cfd);
}

